# -*- coding:utf-8 -*-
import config.MongoConnection as mongo;

db = mongo.getDB();

base_data = db.base_data;  # 基础信息数据
company_data = db.company_data; # 公司基本数据
daily = db.daily; # 日线行情

'''
save stock base data
参数
名称 	类型 	描述
ts_code 	str 	TS代码
symbol 	str 	股票代码
name 	str 	股票名称
area 	str 	所在地域
industry 	str 	所属行业
fullname 	str 	股票全称
enname 	str 	英文全称
market 	str 	市场类型 （主板/中小板/创业板）
exchange_id 	str 	交易所代码
curr_type 	str 	交易货币
list_status 	str 	上市状态： L上市 D退市 P暂停上市
list_date 	str 	上市日期
delist_date 	str 	退市日期
is_hs 	str 	是否沪深港通标的，N否 H沪股通 S深股通
'''
def stockBase(stks):
  if not isinstance(stks, (list)):
    raise TypeError("need a list type.");
  base_data.insert(stks);
  print("save below data to db success:\n", stks)
  return;

'''
save company data
参数
名称 	类型 	默认显示 	描述
ts_code 	str 	Y 	股票代码
chairman 	str 	Y 	法人代表
manager 	str 	Y 	总经理
secretary 	str 	Y 	董秘
reg_capital 	float 	Y 	注册资本
setup_date 	str 	Y 	注册日期
province 	str 	Y 	所在省份
city 	str 	Y 	所在城市
introduction 	str 	Y 	公司介绍
website 	str 	Y 	公司主页
email 	str 	Y 	电子邮件
office 	str 	Y 	办公室
employees 	int 	Y 	员工人数
main_business 	str 	Y 	主要业务及产品
business_scope 	str 	Y 	经营范围
'''
def stockCompany(companys):
  if not isinstance(companys, (list)):
    raise TypeError("need a list type.");
  company_data.insert(companys);
  print("save below data to db success:\n", companys)
  return;

'''
日线行情参数
名称 	类型 	描述
ts_code 	str 	股票代码
trade_date 	str 	交易日期
open 	float 	开盘价
high 	float 	最高价
low 	float 	最低价
close 	float 	收盘价
pre_close 	float 	昨收价
change 	float 	涨跌额
pct_change 	float 	涨跌幅
vol 	float 	成交量 （手）
amount 	float 	成交额 （千元）
'''
def stockDaily(dailys):
  if not isinstance(dailys, (list)):
    raise TypeError("need a list type.");
  daily.insert(dailys);
  print("save below data to db success:\n", dailys)
  return;